package mysql

import (
	"context"
	"fmt"
	"github/kadycui/gids/conf"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
	"log"
	"os"
	"time"
)

var _db *gorm.DB
var (
	mysqlConfig = mysql.Config{
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据版本自动配置
	}
	newLogger = logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
		logger.Config{
			SlowThreshold:             time.Second,   // 慢 SQL 阈值
			LogLevel:                  logger.Silent, // Log level
			Colorful:                  true,          // 彩色打印
			IgnoreRecordNotFoundError: true,          // 忽略记录为空的错误
		},
	)
	createBatchSize = 200 // 批量插入条数
)

// InitMySQL 初始化读.
func InitMySQL() {
	//配置MySQL连接参数
	username := conf.App.MySQL.MySQLUser     //账号
	password := conf.App.MySQL.MySQLPassword //密码
	host := conf.App.MySQL.MySQLHost         //数据库地址，可以是Ip或者域名
	port := conf.App.MySQL.MySQLPort         //数据库端口
	Dbname := conf.App.MySQL.MySQLDb         //数据库名
	dsn := username + ":" + password + "@tcp(" + host + ":" + port + ")/" + Dbname + "?charset=utf8mb4&parseTime=True&loc=Local"
	mysqlConfig.DSN = dsn
	_db, err := gorm.Open(mysql.New(mysqlConfig), &gorm.Config{
		Logger: newLogger,
		NamingStrategy: schema.NamingStrategy{
			TablePrefix:   "",   // 表名前缀
			SingularTable: true, // 禁用表明复数
		}})
	if err != nil {
		if err != nil {
			panic("连接数据库失败, error=" + err.Error())
		}
	}
	sqlDB, _ := _db.DB()
	sqlDB.SetMaxIdleConns(100)
	sqlDB.SetMaxOpenConns(20)
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err = sqlDB.PingContext(ctx); err != nil {
		if err != nil {
			panic("Ping MySQL, error=" + err.Error())
		}
	}
	fmt.Println("MYSQL init OK")

}
func GetDB() *gorm.DB {
	return _db
}
